package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSONArray;
import com.heima.aliyun.GreenImageScan;
import com.heima.aliyun.GreenTextScan;
import com.heima.common.constants.message.PublishArticleConstants;
import com.heima.common.constants.wemedia.WemediaConstants;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import io.seata.common.util.CollectionUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {

    @Autowired
    WmNewsMapper wmNewsMapper;

    //得到webSite
    @Value("${file.oss.web-site}")
    String webSite;


    /**
     * 文章自动审核
     *
     * @param id 自媒体文章id
     */
    @Override
    public void autoScanWmNews(Integer id) {
        log.info(" 自动审核发布方法 被调用   当前审核发布的文章id==> {}", id);
        //检查参数
        if (id == null) {
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID);
        }
        //1. 根据文章id 远程调用feign查询文章
        WmNews wmNews = wmNewsMapper.selectById(id);
        if (wmNews == null) {
            log.error(" 自动审核失败 未查询到该文章 ==> 文章id : {}", id);
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST, "没有该文章");
        }
        // 2. 判断文章状态是否为待审核状态
        Short status = wmNews.getStatus();
        if (status.shortValue() == WemediaConstants.WM_NEWS_SUMMIT_STATUS) {
            // 抽取出文章中 所有的文本内容 和 所有的图片url集合    Map<String,Object>  content 内容   images List<String>
            Map<String, Object> contentMap = handleTextAndImages(wmNews);
            // 3.1  敏感词审核    失败   修改文章状态(2)
            boolean sentiveresult = handleSentive((String) contentMap.get("content"), wmNews);

            if (!sentiveresult) {//如果不为true则直接结束方法
                return;
            }
            //到这步敏感词通过
            log.info(" 自管理敏感词审核通过  =======   ");

            // 3.2  阿里云的文本审核   失败  状态2  不确定 状态3
            boolean isTextScan = handleTextScan((String) contentMap.get("content"), wmNews);
            if (!isTextScan) {//如果不为true则直接结束方法
                return;
            }
            log.info(" 阿里云文本内容审核通过  =======   ");
            // 3.3  阿里云的图片审核   失败  状态2  不确定 状态3
            List<String> images = (List<String>) contentMap.get("images");
            if (CollectionUtils.isNotEmpty(images)) {
                boolean isImageScan = handleImageScan((List<String>) images, wmNews);
                //判断图片是否通过审核
                if (!isImageScan) return;
                log.info(" 阿里云图片审核通过  =======   ");
            }
        }
        // 4. 判断文章发布时间是否大于当前时间   状态 8
        //修改审核后文章的状态
        updateWmNews(wmNews, WmNews.Status.SUCCESS.getCode(), "审核成功!");
        //5. 通知定时发布文章
        //获取发布时间
        long publishTime = wmNews.getPublishTime().getTime();
        //获取当前时间
        long currentTime = System.currentTimeMillis();
        //发布时间减去当前时间
        long remainTime = publishTime - currentTime;

        //利用rabbitTemplate发送消息
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE,
                PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY,
                wmNews.getId(),
                (message) -> {                              // 延时消息 必设置
                    message.getMessageProperties()
                            .setHeader("x-delay", remainTime <= 0 ? 0 : remainTime);
                    return message;   }
        );
        log.info("立即发布文章通知成功发送，文章id : {}", wmNews.getId());
    }

    @Autowired
    RabbitTemplate rabbitTemplate;

    /**
     * 修改文章状态
     *
     * @param wmNews
     * @param status
     * @param reason
     */
    private void updateWmNews(WmNews wmNews, short status, String reason) {
        //修改文章的状态
        wmNews.setStatus(status);
        //修改理由
        wmNews.setReason(reason);
        //通过mapper调用修改方法  最终完成修改
        wmNewsMapper.updateById(wmNews);
    }

    //注入图片审核
    @Autowired
    GreenImageScan greenImageScan;

    /**
     * 阿里云审核图片
     *
     * @param images
     * @param wmNews
     * @return
     */
    private boolean handleImageScan(List<String> images, WmNews wmNews) {
        //创建一个布尔变量 默认值为true
        boolean flag = true;
        try {
            //利用GreenTextScan 中的方法审核文本内容 (利用try catch  包裹代码 以防报错)
            Map map = greenImageScan.imageUrlScan(images);
            //从得到的map集合中获取 suggestion为key的value 转为String类型
            String suggestion = (String) map.get("suggestion");
            //利用switch判断suggestion的值  一共有三种状态  block 违规  pass 通过 review 不确定转为人工审核
            switch (suggestion) {
                case "block":
                    flag = false;
                    updateWmNews(wmNews, WmNews.Status.FAIL.getCode(), "图片中有违规内容,审核未通过");
                    break;
                case "review":
                    flag = false;
                    updateWmNews(wmNews, WmNews.Status.ADMIN_AUTH.getCode(), "图片中有不确定内容,转为人工审核");
                    break;
                default: //返回布尔变量

            }
        } catch (Exception e) {
            e.printStackTrace();
            //如果失败, 则将布尔变量设置为false
            log.error("阿里云图片审核出现异常, 原因: {}" + e.getMessage());
            updateWmNews(wmNews, WmNews.Status.ADMIN_AUTH.getCode(), "阿里云图片服务异常，转为人工审核");
            flag = false;
        }
        return flag;
    }


    //注入文本审核
    @Autowired
    GreenTextScan greenTextScan;

    /**
     * 阿里云文本审核
     *
     * @param content 文本内容
     * @param wmNews  文章信息
     * @return
     */
    private boolean handleTextScan(String content, WmNews wmNews) {
        //创建一个布尔变量 默认值为true
        boolean flag = true;
        try {
            //利用GreenTextScan 中的方法审核文本内容 (利用try catch  包裹代码 以防报错)
            Map map = greenTextScan.greenTextScan(content);
            //从得到的map集合中获取 suggestion为key的value 转为String类型
            String suggestion = (String) map.get("suggestion");
            //利用switch判断suggestion的值  一共有三种状态  block 违规  pass 通过 review 不确定转为人工审核
            switch (suggestion) {
                case "block":
                    updateWmNews(wmNews, WmNews.Status.FAIL.getCode(), "文本中有违规内容,审核未通过");
                    break;
                case "review":
                    updateWmNews(wmNews, WmNews.Status.ADMIN_AUTH.getCode(), "文章中有不确定内容,转为人工审核");
                    break;
                default:
            }
        } catch (Exception e) {
            e.printStackTrace();
            //如果失败, 则将布尔变量设置为false
            log.error("阿里云文本审核出现异常, 原因: {}" + e.getMessage());
            updateWmNews(wmNews, WmNews.Status.ADMIN_AUTH.getCode(), "阿里云内容服务异常，转为人工审核");
            flag = false;
        }
        //返回布尔变量
        return flag;
    }

    //注入admin远程客户端
    @Autowired
    AdminFeign adminFeign;

    /**
     * 敏感词审核
     * 使用远程客户端查询敏感词列表
     *
     * @param content 文本内容
     * @param wmNews  文章信息
     * @return
     */
    private boolean handleSentive(String content, WmNews wmNews) {
        //设置一个布尔变量  默认值为true
        boolean flag = true;
        //查询出数据库中的敏感词
        ResponseResult<List<String>> result = adminFeign.sensitives();
        //判断状态是否不等于0 否则报错
        if (result.getCode().intValue() != 0) {
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR, result.getErrorMessage());
        }
        //拿出结果中的数据得到敏感词集合
        List<String> sensitives = result.getData();
        //将敏感词集合转发DFA数据模型
        SensitiveWordUtil.initMap(sensitives);
        //检测敏感词
        Map<String, Integer> map = SensitiveWordUtil.matchWords(content);
        //判断得到的map集合是否为null 或者 集合长度是否大于0
        if (map != null && map.size() > 0) {
            //将文章的状态改为2 ,2 为审核不通过
            updateWmNews(wmNews, WmNews.Status.FAIL.getCode(), "内容中包含敏感词内容!" + map);
            //将布尔变量值改为false
            flag = false;
        }
        //返回布尔变量
        return flag;
    }

    /**
     * 抽取出文章中所有的文本内容,所有图片的url集合
     *
     * @param wmNews
     * @return
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {
        //拿出文本内容
        String contentJson = wmNews.getContent();
        //判断是否为空
        if (StringUtils.isBlank(contentJson)) {
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID);
        }
        //将文本内容转为List<Map>集合
        List<Map> contentList = JSONArray.parseArray(contentJson, Map.class);

        //抽取文章所有的文本
        String content = contentList.stream()
                //先进行过滤
                .filter(map -> "text".equals(map.get("type")))
                //抽取 转为String 字符串
                .map(map -> (String) map.get("value"))
                //收集成一个字符串
                .collect(Collectors.joining("_hmtt_"));
        //将标题加上中间分隔符
        content = content + "_hmtt_" + wmNews.getTitle();

        //抽取文章中所有的图片
        List<String> images = contentList.stream()
                .filter(map -> "image".equals(map.get("type")))
                .map(map -> (String) map.get("value"))
                .collect(Collectors.toList());

        //判断图片地址集合是否为空
        if (StringUtils.isNotBlank(wmNews.getImages())) {
            //按照逗号切割封面字符串  得到数组 基于数组得到stream  将每一条数据拼接一个前缀  最后收集成一个集合
            List<String> urlList = Arrays.stream(wmNews.getImages().split(","))
                    //抽取
                    .map(url -> webSite + url)
                    //收集成一个集合
                    .collect(Collectors.toList());
            //将图片地址添加到images集合中
            images.addAll(urlList);
        }
        //去重
        images = images.stream().distinct().collect(Collectors.toList());

        //创建一个map集合
        Map<String, Object> map = new HashMap<>();
        //设置文本内容  key content  value 文章文本
        map.put("content", content);
        //设置图片内容  key  images   value 图片集合
        map.put("images", images);
        //返回集合
        return map;
    }
}